
import torch
import torchvision
print(torch.__version__)
print("GPU:",torch.cuda.is_available())
